package BinaryTree;//给定一个二叉搜索树的根节点 root 和一个值 key，删除二叉搜索树中的 key 对应的节点，并保证二叉搜索树的性质不变。返回二叉搜索树（有可能被更新）的
//根节点的引用。 
//
// 一般来说，删除节点可分为两个步骤： 
//
// 
// 首先找到需要删除的节点； 
// 如果找到了，删除它。 
// 
//
// 
//
// 示例 1: 
//
// 
//
// 
//输入：root = [5,3,6,2,4,null,7], key = 3
//输出：[5,4,6,2,null,null,7]
//解释：给定需要删除的节点值是 3，所以我们首先找到 3 这个节点，然后删除它。
//一个正确的答案是 [5,4,6,2,null,null,7], 如下图所示。
//另一个正确答案是 [5,2,6,null,4,null,7]。
//
//
// 
//
// 示例 2: 
//
// 
//输入: root = [5,3,6,2,4,null,7], key = 0
//输出: [5,3,6,2,4,null,7]
//解释: 二叉树不包含值为 0 的节点
// 
//
// 示例 3: 
//
// 
//输入: root = [], key = 0
//输出: [] 
//
// 
//
// 提示: 
//
// 
// 节点数的范围 [0, 10⁴]. 
// -10⁵ <= Node.val <= 10⁵ 
// 节点值唯一 
// root 是合法的二叉搜索树 
// -10⁵ <= key <= 10⁵ 
// 
//
// 
//
// 进阶： 要求算法时间复杂度为 O(h)，h 为树的高度。 
// Related Topics 树 二叉搜索树 二叉树 👍 568 👎 0


//leetcode submit region begin(Prohibit modification and deletion)
/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class deleteNode {
    public TreeNode deleteNode(TreeNode root, int key) {
        /**
         * 第一种情况：没找到删除的节点，遍历到空节点直接返回了
         *         若找到删除的节点
         *             第二种情况：左右孩子都为空（叶子节点），直接删除节点， 返回NULL为根节点
         *             第三种情况：删除节点的左孩子为空，右孩子不为空，删除节点，右孩子补位，返回右孩子为根节点
         *             第四种情况：删除节点的右孩子为空，左孩子不为空，删除节点，左孩子补位，返回左孩子为根节点
         *             第五种情况：左右孩子节点都不为空，则将删除节点的左子树头结点（左孩子）放到删除节点的右子树的最左面节点的左孩子上，
         *               返回删除节点右孩子为新的根节点。
         * */
        return deleteHelper(root, key);


    }
    TreeNode deleteHelper(TreeNode root, int val){
//        第一种情况：没找到删除的节点，遍历到空节点直接返回了
        if(root == null){
            return null;
        }
        //没找到时根据方向去找
        if(root.val > val){
            root.left = deleteHelper(root.left, val);
        }else if(root.val < val){
            root.right = deleteHelper(root.right, val);
        }else{
            //找到时,第三四种情况
            if(root.left==null){
                return root.right;
            }
            if(root.right==null){
                return root.left;
            }
            //若左右孩子节点都不为空，则将删除节点的左子树头结点（左孩子）放到删除节点的右子树的最左面节点的左孩子上
            TreeNode tmp = root.right;
            //找到右子树的最左面节点
            while(tmp.left != null){
                tmp = tmp.left;
            }
            root.val = tmp.val;
            root.right = deleteHelper(root.right,tmp.val);
        }
        return root;
    }
}
//leetcode submit region end(Prohibit modification and deletion)
